package com.future;

import java.util.Arrays;

/**
 * Description:
 * 给你一个长度为n的整数数组nums，其中n > 1，返回输出数组output，其中 output[i]等于nums中除nums[i]之外其余各元素的乘积。
 * <p>
 * <p>
 * 示例:
 * <p>
 * 输入: [1,2,3,4]
 * 输出: [24,12,8,6]
 *  
 * <p>
 * 提示：题目数据保证数组之中任意元素的全部前缀元素和后缀（甚至是整个数组）的乘积都在 32 位整数范围内。
 * <p>
 * 说明: 请不要使用除法，且在O(n) 时间复杂度内完成此题。
 * <p>
 * 进阶：
 * 你可以在常数空间复杂度内完成这个题目吗？（ 出于对空间复杂度分析的目的，输出数组不被视为额外空间。）
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/product-of-array-except-self
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 *
 * @author weiruibai.vendor
 * Date: 2021/9/29 09:50
 */
public class Solution_productExceptSelf_238 {

    public static void main(String[] args) {
        int[] nums = {0, 2, 3, 4, 2};
        int[] s = productExceptSelf(nums);
        System.out.println(Arrays.toString(s));

    }

    /**
     * 超时
     *
     * @param nums
     * @return
     */
    public static int[] productExceptSelf_(int[] nums) {
        int result[] = new int[nums.length];
        for (int i = 0; i < nums.length; i++) {
            int mult = 1;
            for (int j = 0; j < nums.length; j++) {
                if (j == i) {
                    continue;
                }
                mult = mult * nums[j];
            }
            result[i] = mult;
        }
        return result;
    }


    /**
     * 思路：
     * 1、计算总的乘积
     * 2、然后根据一定条件处以当前坐标对应的值就是当前坐标结果的值
     * <p>
     * 注意：
     * 场景1：数组中有两个0以上的场景 [1,2,0,2,1,0]
     * 此时结果都是0，即结果[0,0,0,0,0,0]，这种情况自然成立，无限特殊处理
     * <p>
     * 场景2：数组中只有一个0的场景 {1,2,0,3,1}
     * 此时除了当前值为0的坐标结果为非0外，其他都是0
     * 即结果为[0,0,6,0,0]
     * 需要特殊处理
     *
     * @param nums
     * @return
     */
    public static int[] productExceptSelf(int[] nums) {
        int mult = 1;
        int zeroCount = 0; // 记录值为0的个数
        int zeroIndex = 0; // 记录只有一个值为0的下标
        /**
         * 计算总的乘积
         */
        for (int i = 0; i < nums.length; i++) {
            if (nums[i] == 0) {
                zeroCount++;
                zeroIndex = i;
            } else {
                mult = mult * nums[i];
            }
        }

        int result[] = new int[nums.length];
        // 当0值的个数大于1，此时返回结果数组都是0
        if (zeroCount > 1) {
            // 直接返回
            return result;
        } else if (zeroCount == 1) {
            // 除了zeroIndex为非0外，其他都是0
            result[zeroIndex] = mult;
            return result;
        }
        for (int i = 0; i < nums.length; i++) {
            if (nums[i] == 0) {
                result[i] = mult;
            } else {
                result[i] = mult / nums[i];
            }
        }
        return result;
    }

}
